﻿# import sys
# sys.path.append(r"D:\codes\PyTorch_nn")
# from .resnet import resnet18, resnet50

# model_dict = {
#     "resnet18": resnet18,
#     "resnet50": resnet50,
# }


# def create_model(model_name, num_classes):
#     return model_dict[model_name](num_classes=num_classes)